Snowflakeのストアドプロシージャを使ってPythonからSnowflakeへSQLクエリを投げてみた

-- 権限が不安な方は実行を簡単にするため強めの権限を持つロールを使用
USE ROLE ACCOUNTADMIN;

-- データベース指定
USE DATABASE <お使いのDB>; -- なければ新規作成してください

-- スキーマから準備
CREATE OR REPLACE SCHEMA test_procedure;

CREATE OR REPLACE TABLE test_procedure_table(col1 varchar(5));

INSERT INTO test_procedure.test_procedure_table VALUES
    ('test1'),('test2'), ('test3'), ('test4'), ('test5');

SELECT * FROM TEST_PROCEDURE.TEST_PROCEDURE_TABLE;

CREATE OR REPLACE PROCEDURE TEST_PROCEDURE(
    db_name string,
    schema_name string,
    table_name string
)
RETURNS STRING
LANGUAGE PYTHON
RUNTIME_VERSION = '3.11'
PACKAGES = ('snowflake-snowpark-python')
HANDLER = 'main'
AS
$$
from snowflake.snowpark import Session

def main(db_name: str, schema_name: str, table_name: str):
    connection_parameters = {
        "account": "<your snowflake account>",
        "user": "<your snowflake user>",
        "password": "<your snowflake password>"
    }
    with Session.builder.configs(connection_parameters).create() as session:
        query = f"SELECT * FROM {db_name}.{schema_name}.{table_name};"
        data = session.sql(query).collect()

    return data
$$
;

-- ストアドプロシージャを呼び出す
CALL TEST_PROCEDURE('<お使いのDB名>', 'test_procedure', 'test_procedure_table');

DROP SCHEMA test_procedure;